package _dp

import org.junit.Test

/**
 * https://leetcode.cn/problems/best-time-to-buy-and-sell-stock/description
 * https://blog.csdn.net/zyx_ly/article/details/104760783
 *
 * 121. 买卖股票的最佳时机
 *
 * 给定一个数组 prices ，它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。
 * 你只能选择 某一天 买入这只股票，并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
 * 返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润，返回 0 。
 *
 * 示例 1：
 * 输入：[7,1,5,3,6,4]
 * 输出：5
 * 解释：在第 2 天（股票价格 = 1）的时候买入，在第 5 天（股票价格 = 6）的时候卖出，最大利润 = 6-1 = 5 。
 *      注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格；同时，你不能在买入前卖出股票。
 *
 * 示例 2：
 * 输入：prices = [7,6,4,3,1]
 * 输出：0
 * 解释：在这种情况下, 没有交易完成, 所以最大利润为 0。
 *
 *
 * 提示：
 * 1 <= prices.length <= 105
 * 0 <= prices[i] <= 104
 */
class leetcode_121 {
    fun maxProfit(prices: IntArray): Int {
        /**
        题型：动态规划

        思路：

        递推公式 ：
        假如第i天卖，dp[i] = max(dp[i-1], prices[i] - minPrice), 但是，若prices 大小很大时，使用dp数组会超时。
        时间复杂度O(N)
        空间复杂度O(N)

        空间优化后：
        假如第i天卖，max = max(max, prices[i] - minPrice)
        时间复杂度O(N)
        空间复杂度O(1)
         */
        return maxProfit1(prices)
//        return maxProfit2(prices)
    }

    fun maxProfit1(prices: IntArray): Int {
        if (prices.size == 1) {
            return 0
        }
        // 1 确定数组以及下标
        // 第i天能获取的利润
        val dp: IntArray = IntArray(prices.size)
        var minPrice: Int = prices[0]

        // 2 确定递推公式
        // 假如第i天卖，dp[i] = max(dp[i-1], prices[i] - minPrice)

        // 3 初始化数组
        // dp[0] = 0

        // 4 遍历顺序
        for (i in 1 until prices.size) {
            // 第i天时，更新最小价格
            if (prices[i] < minPrice) {
                minPrice = prices[i]
            }
            // 假如第i天卖，取dp[i-1]与 第i天卖的利润最大值
            dp[i] = Math.max(dp[i - 1], prices[i] - minPrice)

            // 5 打印数组
            println("i=$i,minPrice=$minPrice,dp[i]=max(${dp[i - 1]}, ${prices[i]} - ${minPrice})=${dp[i]}")
        }
        return dp[dp.size - 1]
    }

    fun maxProfit2(prices: IntArray): Int {
        if (prices.size == 1) {
            return 0
        }
        // 1 确定数组以及下标
        var max: Int = 0 // 第i天能获取的利润
        var minPrice: Int = prices[0]

        // 2 确定递推公式
        // 假如第i天卖，max = max(max, prices[i] - minPrice)

        // 3 初始化数组

        // 4 遍历顺序
        for (i in 1 until prices.size) {
            // 第i天时，更新最小价格
            if (prices[i] < minPrice) {
                minPrice = prices[i]
            }
            // 假如第i天卖，取 旧max与 第i天卖的利润最大值
            max = Math.max(max, prices[i] - minPrice)

            // 5 打印数组
            println("i=$i,minPrice=$minPrice,max(${max}, ${prices[i]} - ${minPrice})=$max")
        }
        return max
    }
}